package screenshot;


import org.apache.commons.io.FileUtils;
import org.junit.jupiter.api.Test;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.chrome.ChromeDriver;

import java.io.File;
import java.io.IOException;
import java.time.Duration;

// 屏幕截图
public class ScreenShotTest {
    @Test
    public void screenshotTest() throws IOException {
        // 打开游览器驱动
        ChromeDriver driver = new ChromeDriver();

        // 进行隐式等待, 最多等待 10 s
        driver.manage().timeouts().implicitlyWait(Duration.ofSeconds(10));

        // 进入 百度一下
        driver.get("https://www.baidu.com/");
        // 截图, 以文件的格式
        File srcFile = driver.getScreenshotAs(OutputType.FILE);
        // 保存的文件路径在项目目录下
        String savePath = "./picture.png";
        FileUtils.copyFile(srcFile, new File(savePath));
        // 关闭游览器
        driver.quit();
    }
}
